import torch
from diffusers import StableDiffusionPipeline

import datetime

start = datetime.datetime.now()
model_id = "C:\\Users\\21790\\lsx\\code\\diffusers\\models\\deliberate_v2.safetensors"
pipe = StableDiffusionPipeline.from_pretrained(model_id,use_safetensors=True,local_files_only=True, torch_dtype=torch.float32).to("cuda")
print(f"Loaded in {datetime.datetime.now() - start}")
image = pipe("example prompt", num_inference_steps=2).images[0]
image.save("example.png")


